import matplotlib.pyplot as plt

# 假设你已经有了这些数据
x_a = [0, 1, 2, 3, 4]
y_a = [10, 20, 30, 40, 50]
x_b = [0, 1, 2, 3, 4]
y_b = [50, 40, 30, 20, 10]

# 创建一个新的图形窗口
plt.figure()

# 在当前图形窗口中创建一个新的子图
plt.subplot(1, 2, 1)

# 使用三角形标记（'^'）和实线绘制第一组数据
plt.plot(x_a, y_a, '^-', label='First plot')

# 使用原点标记（'o'）和虚线绘制第二组数据
plt.plot(x_b, y_b, 'o-', label='Second plot')

# 添加图例
plt.legend()

# 设置y轴的范围
plt.ylim([0, 100])

# 设置x和y轴的标签以及标题
plt.xlabel('Episode')
plt.ylabel('Reward')
plt.title('4.2a |U|=3,|K|=2')

# 对第二个子图重复上述步骤
plt.subplot(1, 2, 2)
plt.plot(x_a, y_a, '^-', label='First plot')
plt.plot(x_b, y_b, 'o-', label='Second plot')
plt.legend()
plt.ylim([0, 100])
plt.xlabel('Episode')
plt.ylabel('Reward')
plt.title('4.2b |U|=2,|K|=2')

# 显示所有的子图
plt.show()